// 监听来自content script的消息
chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
  if (message.type === 'keyPress') {
    // 获取按键名称
    let keyName = message.key;
    
    // 对于特殊键进行格式化显示
    if (keyName === ' ') keyName = 'Space';
    if (keyName === 'Control') keyName = 'Ctrl';
    if (keyName === 'ArrowUp') keyName = '↑';
    if (keyName === 'ArrowDown') keyName = '↓';
    if (keyName === 'ArrowLeft') keyName = '←';
    if (keyName === 'ArrowRight') keyName = '→';
    
    // 从存储中获取当前统计数据
    chrome.storage.local.get('keyStats', (result) => {
      const keyStats = result.keyStats || {};
      
      // 更新按键计数
      keyStats[keyName] = (keyStats[keyName] || 0) + 1;
      
      // 保存更新后的统计数据
      chrome.storage.local.set({ keyStats: keyStats });
    });
  }
});
